package ai.hou.camera.util;


import android.graphics.Bitmap;
import android.graphics.Color;
import android.text.TextUtils;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.BinaryBitmap;
import com.google.zxing.DecodeHintType;
import com.google.zxing.EncodeHintType;
import com.google.zxing.RGBLuminanceSource;
import com.google.zxing.Result;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.common.HybridBinarizer;
import com.google.zxing.qrcode.QRCodeReader;
import com.google.zxing.qrcode.QRCodeWriter;

import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;

import io.reactivex.Observable;
import io.reactivex.android.schedulers.AndroidSchedulers;
import io.reactivex.annotations.NonNull;
import io.reactivex.functions.Function;
import io.reactivex.observers.DisposableObserver;
import io.reactivex.schedulers.Schedulers;

/**
 * zxing工具类
 *
 * @author zhaoxiaolei
 */
public class ZxingUtils {
    private static final String TAG = "ZxingUtils";

    public static Bitmap createQRCodeBitmap(String content, int w, int h, int margin) {
        return createQRCodeBitmap(content, w, h, "UTF-8", "L", margin, Color.BLACK, Color.WHITE);
    }

    /**
     * 生成简单二维码
     *
     * @param content                字符串内容
     * @param width                  二维码宽度
     * @param height                 二维码高度
     * @param character_set          编码方式（一般使用UTF-8）
     * @param error_correction_level 容错率 L：7% M：15% Q：25% H：35%
     * @param margin                 空白边距（二维码与边框的空白区域）
     * @param color_black            黑色色块
     * @param color_white            白色色块
     * @return BitMap
     */
    private static Bitmap createQRCodeBitmap(String content, int width, int height,
                                             String character_set, String error_correction_level,
                                             int margin, int color_black, int color_white) {
        // 字符串内容判空
        if (TextUtils.isEmpty(content)) {
            return null;
        }
        // 宽和高>=0
        if (width < 0 || height < 0) {
            return null;
        }
        try {
            /** 1.设置二维码相关配置 */
            Hashtable<EncodeHintType, String> hints = new Hashtable<>();
            // 字符转码格式设置
            if (!TextUtils.isEmpty(character_set)) {
                hints.put(EncodeHintType.CHARACTER_SET, character_set);
            }
            // 容错率设置
            if (!TextUtils.isEmpty(error_correction_level)) {
                hints.put(EncodeHintType.ERROR_CORRECTION, error_correction_level);
            }
            // 空白边距设置
//            if (margin > 0) {
            hints.put(EncodeHintType.MARGIN, String.valueOf(margin));
//            }
            /** 2.将配置参数传入到QRCodeWriter的encode方法生成BitMatrix(位矩阵)对象 */
            BitMatrix bitMatrix = new QRCodeWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);

            /** 3.创建像素数组,并根据BitMatrix(位矩阵)对象为数组元素赋颜色值 */
            int[] pixels = new int[width * height];
            for (int y = 0; y < height; y++) {
                for (int x = 0; x < width; x++) {
                    //bitMatrix.get(x,y)方法返回true是黑色色块，false是白色色块
                    if (bitMatrix.get(x, y)) {
                        pixels[y * width + x] = color_black;//黑色色块像素设置
                    } else {
                        pixels[y * width + x] = color_white;// 白色色块像素设置
                    }
                }
            }
            /** 4.创建Bitmap对象,根据像素数组设置Bitmap每个像素点的颜色值,并返回Bitmap对象 */
            Bitmap bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
            bitmap.setPixels(pixels, 0, width, 0, 0, width, height);
            return bitmap;
        } catch (WriterException e) {
            e.printStackTrace();
            return null;
        }
    }

    public interface DecodeQrCodeListener {
        void onSuccess(String text);

        void onFailed();
    }

    public static void decodeQrCode(Bitmap bmp, DecodeQrCodeListener listener) {
        if (listener == null) {
            return;
        }
        int width = bmp.getWidth();
        int height = bmp.getHeight();
        int[] data = new int[width * height];
        bmp.getPixels(data, 0, width, 0, 0, width, height);
        decodeQrCode(data, width, height, listener);
    }

    public static void decodeQrCode(int[] data, int width, int height, DecodeQrCodeListener listener) {
        if (listener == null) {
            return;
        }
        RGBLuminanceSource source = new RGBLuminanceSource(width, height, data);
        BinaryBitmap binaryBitmap = new BinaryBitmap(new HybridBinarizer(source));

        decodeQrCode(binaryBitmap, listener);
    }


    public static void decodeQrCode(BinaryBitmap binaryBitmap, DecodeQrCodeListener listener) {
        Observable.just(binaryBitmap)
                .map((Function<BinaryBitmap, Result>) bitmap -> {
                    Map<DecodeHintType, Object> hints = new HashMap<>();
                    hints.put(DecodeHintType.TRY_HARDER, Boolean.TRUE);
                    hints.put(DecodeHintType.CHARACTER_SET, "UTF-8");
                    QRCodeReader qrCodeReader = new QRCodeReader();
                    return qrCodeReader.decode(bitmap, hints);
                })
                .subscribeOn(Schedulers.newThread())
                .observeOn(AndroidSchedulers.mainThread())
                .subscribe(new DisposableObserver<Result>() {
                    @Override
                    public void onNext(@NonNull Result result) {
                        listener.onSuccess(result.getText());
                    }

                    @Override
                    public void onError(@NonNull Throwable e) {
                        e.printStackTrace();
                        dispose();
                        listener.onFailed();
                    }

                    @Override
                    public void onComplete() {
                        dispose();
                    }
                });
    }
}
